<template>
  <el-dialog
    title="分配菜单"
    :visible.sync="visible"
    :close-on-click-modal="false"
    :before-close="handleCloseDialog"
    width="600px"
  >
    <el-tree
      :data="menuTreeList"
      show-checkbox
      default-expand-all
      node-key="id"
      ref="tree"
      highlight-current
      :props="defaultProps"
    >
    </el-tree>
    <span slot="footer" class="dialog-footer">
      <el-button type="primary" @click="handleSave()" size="mini">保存</el-button>
      <el-button @click="handleClear()" size="mini">清空</el-button>
    </span>
  </el-dialog>
</template>
<script>
import { fetchTreeList } from "@/api/menu";
import { listMenuByRole, allocMenu } from "@/api/role";

export default {
  name: "allocMenuDialog",
  data() {
    return {
      menuTreeList: [],
      defaultProps: {
        children: "children",
        label: "title"
      }
    };
  },
  props: {
    visible: {
      type: Boolean,
      default: false
    },
    roleId: {
      type: String,
      default: ""
    }
  },
  mounted() {
    this.treeList();
    this.getRoleMenu(this.roleId);
  },
  methods: {
    handleCloseDialog() {
      this.$emit("update:visible");
    },
    treeList() {
      fetchTreeList().then(response => {
        this.menuTreeList = response.data;
      });
    },
    getRoleMenu(roleId) {
      listMenuByRole(roleId).then(response => {
        let menuList = response.data;
        let checkedMenuIds = [];
        if (menuList != null && menuList.length > 0) {
          for (let i = 0; i < menuList.length; i++) {
            let menu = menuList[i];
            if (menu.parentId !== 0) {
              checkedMenuIds.push(menu.id);
            }
          }
        }
        this.$refs.tree.setCheckedKeys(checkedMenuIds);
      });
    },
    handleSave() {
      let checkedNodes = this.$refs.tree.getCheckedNodes();
      let checkedMenuIds = new Set();
      if (checkedNodes != null && checkedNodes.length > 0) {
        for (let i = 0; i < checkedNodes.length; i++) {
          let checkedNode = checkedNodes[i];
          checkedMenuIds.add(checkedNode.id);
          if (checkedNode.parentId !== 0) {
            checkedMenuIds.add(checkedNode.parentId);
          }
        }
      }
      this.$confirm("是否分配菜单？", "提示", {
        confirmButtonText: "确定",
        cancelButtonText: "取消",
        type: "warning"
      }).then(() => {
        
        let params = new URLSearchParams();
        params.append("roleId", this.roleId);
        params.append("menuIds", Array.from(checkedMenuIds));
        allocMenu(params).then(response => {
          this.$message({
            message: "分配成功",
            type: "success",
            duration: 1000
          });
          this.handleCloseDialog();
        });
      });
    },
    handleClear() {
      this.$refs.tree.setCheckedKeys([]);
    }
  }
};
</script>
